#!/bin/bash 
#SBATCH -J task_lambda
#SBATCH -p gpu-normal 
#SBATCH -N 1 
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH -o task_lambda.out 
#SBATCH -e task_lambda.err


module load anaconda3/2019.10 cuda/10.1

cd ..
pwd

python train.py --gpu_id=0 --output_dir=output_20 --is_parallel=False --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=48  --seq_len=12 --w_lambda=0 --delta_type=time_delta --layer_size=z_dim --is_fc=True --is_ln=True --is_res=True --impute_iter=2000

python train.py --gpu_id=0 --output_dir=output_20 --is_parallel=False --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=48  --seq_len=12 --w_lambda=0.125 --delta_type=time_delta --layer_size=z_dim --is_fc=True --is_ln=True --is_res=True --impute_iter=2000

python train.py --gpu_id=0 --output_dir=output_20 --is_parallel=False --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=48  --seq_len=12 --w_lambda=0.25 --delta_type=time_delta --layer_size=z_dim --is_fc=True --is_ln=True --is_res=True --impute_iter=2000 

python train.py --gpu_id=0 --output_dir=output_20 --is_parallel=False --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=48  --seq_len=12 --w_lambda=0.5 --delta_type=time_delta --layer_size=z_dim --is_fc=True --is_ln=True --is_res=True --impute_iter=2000   
python train.py --gpu_id=0 --output_dir=output_20 --is_parallel=False --is_load=True --iteration=5000 --log_interval=100 --gru_type=grui --num_layer=2 --hidden_dim=48  --seq_len=12 --w_lambda=1 --delta_type=time_delta --layer_size=z_dim --is_fc=True --is_ln=True --is_res=True --impute_iter=2000   
